# -*- coding:utf-8 -*-
# 删除文件夹
import os
import shutil

# 需要删除的文件夹前缀，即已values-开头的文件夹
DIR_PRE = 'values-'


# 删除当前目录下的指定文件夹
def delete_current_dir():
    work_dir = os.getcwd()
    delete_dir(work_dir)


# 递归删除所有指定名称的文件夹
def delete_dir(root_dir):
    # 判断目录是否是文件夹
    if os.path.isdir(root_dir):
        # 获取文件夹下所有的子文件夹
        child_dirs = os.listdir(root_dir)
        # 遍历所有子文件夹
        for child_dir in child_dirs:
            # 判断文件夹的名称是否包含 dir_pre 字段
            if DIR_PRE in child_dir:
                # 删除文件夹
                shutil.rmtree(os.path.join(root_dir, child_dir))
            else:
                # 递归查找
                delete_dir(os.path.join(root_dir, child_dir))
    else:
        return


if __name__ == "__main__":
    delete_current_dir()
